package xishuiwanglearning.controller;

import org.apache.commons.io.FileUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.util.MultiValueMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.io.*;

@Controller
public class FileUpandDownController {
    //下载文件
    @RequestMapping("/testDown")

    public ResponseEntity<byte[]> testResponseEntity(HttpSession session) throws IOException {
       //获取SerletContext对象
        ServletContext servletContext = session.getServletContext();
        //获取服务器中文件的真实路径
        String realPath = servletContext.getRealPath("/static/img/1.png");
        //创建输入流
        InputStream is = new FileInputStream(realPath);
        //创建字节数组
        byte[] bytes = new byte[is.available()];
        //将流读入到字节数组中
        is.read(bytes);
        //创建HttpHeader对象设置响应头信息
        MultiValueMap<String,String> headers = new HttpHeaders();
        //设置要下载的方式以及下载文件的名字
        headers.add("Content-Disposition","attachment;filename=1.png");
        //设置相应代码
        HttpStatus statusCode = HttpStatus.OK;
        //创建ResponseEntity对象
        ResponseEntity<byte[]> responseEntity  = new ResponseEntity<>(bytes,headers,statusCode);
        //关闭输入流
        is.close();
        //返回值
        return responseEntity;
    }
        @RequestMapping("/testUp")
        public String testUp(MultipartFile photo,HttpSession session) throws IOException {

        String fileName = photo.getOriginalFilename();
        ServletContext servletContext = session.getServletContext();
        String photoPath = servletContext.getRealPath("photo");
        File file = new File(photoPath);
        //判断photopath对应的路径是否存在
            if(!file.exists())
            {
                file.mkdir();
            }

            String finalpath = photoPath + File.separator + fileName;

            photo.transferTo(new File(finalpath));
            System.out.println(photo.getName());
            System.out.println(photo.getOriginalFilename());
            return "success";
        }

}
